Data science interview question